//
// Created by Administrator on 2021/4/17.
//
/*

给你一个整数数组 nums 和两个整数k 和 t 。请你判断是否存在两个下标 i 和 j，使得abs(nums[i] - nums[j]) <= t ，同时又满足 abs(i - j) <= k 。

如果存在则返回 true，不存在返回 false。


示例1：

输入：nums = [1,2,3,1], k = 3, t = 0
输出：true
示例 2：

输入：nums = [1,0,1,1], k = 1, t = 2
输出：true
示例 3：

输入：nums = [1,5,9,1,5,9], k = 2, t = 3
输出：false


提示：

0 <= nums.length <= 2 * 104
-231 <= nums[i] <= 231 - 1
0 <= k <= 104
0 <= t <= 231 - 1

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/contains-duplicate-iii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。*/
#include <vector>
#include <iostream>
#include <set>
#include <climits>

using namespace std;

class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int> &&nums, int k, int t) {
        // 滑动窗口
        int n = nums.size();
        set<int> rec;
        for (int i = 0; i < n; i++) {
            // 判断集合中大于x-t的最小值是否小于x+t
            // INT_MIN INT_MAX 防止溢出
//            auto iter = rec.lower_bound(max(nums[i], INT_MIN + t) - t);
            auto iter = rec.lower_bound(nums[i] - t);
//            if (iter != rec.end() && *iter <= min(nums[i], INT_MAX - t) + t) {
            if (iter != rec.end() && *iter <= nums[i] + t) {
                return true;
            }
            // 将x插入set
            rec.insert(nums[i]);
            // 删除第一个元素
            if (i >= k) {
                rec.erase(nums[i - k]);
            }
        }
        return false;
    }

};

int main() {
    Solution sol;
    cout << (sol.containsNearbyAlmostDuplicate(vector<int>{1, 2, 3, 1}, 3, 0) ? "true" : "false");
    cout << (sol.containsNearbyAlmostDuplicate(vector<int>{1, 0, 1, 1}, 1, 2) ? "true" : "false");
    cout << (sol.containsNearbyAlmostDuplicate(vector<int>{1, 5, 9, 1, 5, 9}, 2, 3) ? "true" : "false");
    cout << (sol.containsNearbyAlmostDuplicate(vector<int>{-2147483648, 2147483647}, 1, 1) ? "true" : "false");
    return 0;
}